package com.kyub.spring.gears.services.impl;

import com.kyub.spring.gears.model.FileLayout;
import com.kyub.spring.gears.services.FileStorageService;

import java.io.*;

/**
 * Sample implememntation of FileStorageService
 */
public class LacalFileStorageService implements FileStorageService {

    private String basePath = "";

    public String getBasePath() {
        return basePath;
    }

    public void setBasePath(String basePath) {
        this.basePath = basePath;
    }

    /**
     * @deprecated
     * @param url of the file from the pojo
     * @return
     * @throws IOException
     */
    public byte[] load(String url) throws IOException {
        return getBytesFromFile(new File(url));
    }

    public void save(final String pojoName, final Serializable id, final String propertyName, final FileLayout fileLayout, final String originalFilename, final byte[] payload) throws IOException {

        System.out.println("originalFilename " + originalFilename);
        String ids = "";
        if (id != null)
            ids = String.valueOf(id);

        final String path = fileLayout.getPersistPath() + "/" + pojoName + "_" + ids + "_" + propertyName + "_" + System.currentTimeMillis() + "." + fileLayout.getExtension();

        FileOutputStream fos = new FileOutputStream(path);
        fos.write(payload);
        fos.close();
    }


    private byte[] getBytesFromFile(File file) throws IOException {
        InputStream is = new FileInputStream(file);

        // Get the size of the file
        final long length = file.length();

        // You cannot create an array using a long type.
        // It needs to be an int type.
        // Before converting to an int type, check
        // to ensure that file is not larger than Integer.MAX_VALUE.
        if (length > Integer.MAX_VALUE) {
            throw new IOException("File Too large");
        }

        // Create the byte array to hold the data
        final byte[] bytes = new byte[(int) length];

        // Read in the bytes
        int offset = 0;
        int numRead = 0;
        while (offset < bytes.length
                && (numRead = is.read(bytes, offset, bytes.length - offset)) >= 0) {
            offset += numRead;
        }

        // Ensure all the bytes have been read in
        if (offset < bytes.length) {
            throw new IOException("Could not completely read file " + file.getName());
        }

        // Close the input stream and return bytes
        is.close();
        return bytes;
    }


}
